"""
@File : config.py
@Description: 参数管理器
@Author: bin.chen
@Contact: LebesgueIntigrade@163.com
@Time: 2021/8/30
@IDE: Pycharm Professional
@REFERENCE: None
"""
import os
import platform


class Config(object):
    """
    参数管理器
    """
    def __init__(self):
        self.device = 'cuda'
        self.inference_device = 'cuda'
        self.bert_path = 'chinese-bert-wwm-ext'
        self.vocab_file = 'chinese-bert-wwm-ext/vocab.txt'
        self.hidden_size = 50  # 隐层维度
        self.embedding_size = 768  # embedding维度
        self.lr = 0.001
        self.epoch = 20
        self.batch_size = 16
        self.datapath = "E:/SoftMaskedBert-main/data/narts/SIGHAN15_train.csv"
        self.checkpoints ='./checkpoints'

    @staticmethod
    def platform():
        """
        Returns:
            返回当前操作系统名称,'Linux'或'Windows'
        """
        return platform.system()





















